package com.youzhong.controller;

import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipOutputStream;

import java.io.*;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.ConcurrentHashMap;
import java.util.zip.CRC32;
import java.util.zip.CheckedOutputStream;

/**
 * 功能:使用Apache Ant里提供的org.apache.tools.zip实现zip压缩和解压 (支持中文文件名)
 * 解决了由于java.util.zip包不支持汉字的问题。 使用java.util.zip包时,当zip文件中有名字为中文的文件时,
 * 就会出现异常:"Exception in thread "main " java.lang.IllegalArgumentException at
 * java.util.zip.ZipInputStream.getUTF8String(ZipInputStream.java:285)
 * 
 * @author 夏明龙 E-mail:邮箱
 * @version 创建时间：2013-3-22 上午10:40:21 类说明：
 */
public class TestFile {
	private static List list = new ArrayList();

	private static List listFile(String path) {
		File file = new File(path);
		String[] array = null;
		String sTemp = "";

		if (!file.isDirectory()) {
			return null;
		}
		array = file.list();
		if (array.length > 0) {
			for (int i = 0; i < array.length; i++) {
				sTemp = path + array[i];
				file = new File(sTemp);
				if (file.isDirectory()) {
					listFile(sTemp + "/");
				} else
					list.add(sTemp);
			}
		} else {
			return null;
		}

		return list;
	}

	public static void zip(String needtozipfilepath, String zipfilepath) {
		try {
			byte[] b = new byte[512];

			File needtozipfile = new File(needtozipfilepath);

			if (!needtozipfile.exists()) {
				System.err.println("指定的要压缩的文件或目录不存在.");
				return;
			}

			String zipFile = zipfilepath;
			File targetFile = new File(zipFile.substring(0, zipFile.indexOf("\\") + 1));

			if (!targetFile.exists()) {
				System.out.println("指定的目标文件或目录不存在.");
				return;
			}

			String filepath = needtozipfilepath;
			List fileList = listFile(filepath);
			FileOutputStream fileOutputStream = new FileOutputStream(zipFile);
			CheckedOutputStream cs = new CheckedOutputStream(fileOutputStream, new CRC32());
			ZipOutputStream out = new ZipOutputStream(new BufferedOutputStream(cs));

			for (int i = 0; i < fileList.size(); i++) {
				InputStream in = new FileInputStream((String) fileList.get(i));
				String fileName = ((String) fileList.get(i)).replace(File.separatorChar, '/');
				fileName = fileName.substring(fileName.indexOf("/") + 1);
				ZipEntry e = new ZipEntry(fileName);
				out.putNextEntry(e);
				int len = 0;
				while ((len = in.read(b)) != -1) {
					out.write(b, 0, len);
				}
				out.closeEntry();
			}
			out.close();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	// ///////////////////////////////////////
	/**
	 * 压缩文件 或者 文件夹
	 * 
	 * @param baseDirName    压缩的根目录
	 * @param fileNames       根目录下待压缩的文件或文件夹名
	 * @param targetFileName 目标ZIP 文件 星号 "*" 表示压缩根目录下的全部文件
	 * 
	 */
	public static boolean zip(String baseDirName, String[] fileNames, String targetFileName, String encoding) {
		boolean flag = false;
		try {
			// 判断 "压缩的根目录"是否存在! 是否是一个文件夹!
			File baseDir = new File(baseDirName);
			if (!baseDir.exists() || (!baseDir.isDirectory())) {
				System.err.println("压缩失败! 根目录不存在: " + baseDirName);
				return false;
			}

			// 得到这个 "压缩的根目录" 的绝对路径
			String baseDirPath = baseDir.getAbsolutePath();

			// 由这个 "目标 ZIP 文件" 文件名得到一个 压缩对象 ZipOutputStream
			File targetFile = new File(targetFileName);
			ZipOutputStream out = new ZipOutputStream(new FileOutputStream(targetFile));
			// 中文有乱码，引进下面的改造类
			// CnZipOutputStream out = new CnZipOutputStream(new
			// FileOutputStream(targetFile),encoding);

			// 设置压缩编码Apache Ant有个包专门处理ZIP文件，可以指定文件名的编码方式。由此可以解决问题。例如：用
			// org.apache.tools.zip.ZipOutputStream代替java.util.zip.ZipOutputStream。ZipOutputStream
			// out = .....; out.setEncoding("GBK");
			// out.setEncoding("GBK");//设置为GBK后在windows下就不会乱码了，如果要放到Linux或者Unix下就不要设置了
			out.setEncoding(encoding);

			// "*" 表示压缩包括根目录 baseDirName 在内的全部文件 到 targetFileName文件下
			if (fileNames.equals("*")) {
				TestFile.dirToZip(baseDirPath, baseDir, out);
			} else {
				File[] files = new File[fileNames.length];
				for (int i = 0; i < files.length; i++) {
					// 根据 parent 抽象路径名和 child 路径名字符串创建一个新 File 实例。
					files[i] = new File(baseDir, fileNames[i]);
				}
				if (files[0].isFile()) {
					// 调用本类的一个静态方法 压缩一个文件
					// CompressUtil.fileToZip(baseDirPath, file, out);
					TestFile.filesToZip(baseDirPath, files, out);
				}

			}
			out.close();
			// System.out.println("压缩成功! 目标文件名为: " + targetFileName);
			flag = true;
		} catch (FileNotFoundException e) {
			e.printStackTrace();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return flag;
	}

	/**
	 * 将文件压缩到Zip 输出流
	 * 
	 * @param baseDirPath 根目录路径
	 * @param file        要压缩的文件
	 * @param out         输出流
	 * @throws IOException
	 */
	private static void fileToZip(String baseDirPath, File file, ZipOutputStream out) throws IOException {
		//
		FileInputStream in = null;
		org.apache.tools.zip.ZipEntry entry = null;
		// 创建复制缓冲区 1024*4 = 4K
		byte[] buffer = new byte[1024 * 4];
		int bytes_read = 0;
		if (file.isFile()) {
			in = new FileInputStream(file);
			// 根据 parent 路径名字符串和 child 路径名字符串创建一个新 File 实例
			String zipFileName = getEntryName(baseDirPath, file);
			entry = new org.apache.tools.zip.ZipEntry(zipFileName);
			// "压缩文件" 对象加入 "要压缩的文件" 对象
			out.putNextEntry(entry);
			// 现在是把 "要压缩的文件" 对象中的内容写入到 "压缩文件" 对象
			while ((bytes_read = in.read(buffer)) != -1) {
				out.write(buffer, 0, bytes_read);
			}
			out.closeEntry();
			in.close();
			// System.out.println("添加文件" + file.getAbsolutePath()+ "被添加到 ZIP
			// 文件中!");
		}
	}

	/**
	 * 多个文件目录压缩到Zip 输出流
	 * 
	 * @param baseDirPath 啊啊
	 * @param files aa
	 * @param out 啊啊
	 * @throws IOException 啊啊
	 */
	@SuppressWarnings("unused")
	private static void filesToZip(String baseDirPath, File[] files, ZipOutputStream out) throws IOException {
		// 遍历所有的文件 一个一个地压缩
		for (int i = 0; i < files.length; i++) {
			File file = files[i];
			if (file.isFile()) {
				// 调用本类的一个静态方法 压缩一个文件
				TestFile.fileToZip(baseDirPath, file, out);
			} else {
				/*
				 * 这是一个文件夹 所以要再次得到它下面的所有的文件 这里是自己调用自己..............递归..........
				 */
				//TestFile.dirToZip(baseDirPath, file, out);
			}
		}
	}

	/**
	 * 将文件目录压缩到Zip 输出流
	 * 
	 * @param baseDirPath
	 * @param dir
	 * @param out
	 * @throws IOException
	 */
	private static void dirToZip(String baseDirPath, File dir, ZipOutputStream out) throws IOException {
		// 得到一个文件列表 (本目录下的所有文件对象集合)
		File[] files = dir.listFiles();
		// 要是这个文件集合数组的长度为 0 , 也就证明了这是一个空的文件夹,虽然没有再循环遍历它的必要,但是也要把这个空文件夹也压缩到目标文件中去
		if (files.length == 0) {
			// 根据 parent 路径名字符串和 child 路径名字符串创建一个新 File 实例
			String zipFileName = getEntryName(baseDirPath, dir);
			org.apache.tools.zip.ZipEntry entry = new org.apache.tools.zip.ZipEntry(zipFileName);
			out.putNextEntry(entry);
			out.closeEntry();
		} else {
			// 遍历所有的文件 一个一个地压缩
			for (int i = 0; i < files.length; i++) {
				File file = files[i];
				if (file.isFile()) {
					// 调用本类的一个静态方法 压缩一个文件
					TestFile.fileToZip(baseDirPath, file, out);
				} else {
					/*
					 * 这是一个文件夹 所以要再次得到它下面的所有的文件 这里是自己调用自己..............递归..........
					 */
					TestFile.dirToZip(baseDirPath, file, out);
				}
			}
		}
	}

	/**
	 * 获取 待压缩文件在 ZIP 文件中的 entry的名字，即相对于根目录的相对路径名
	 * 
	 * @param baseDirPath 根目录
	 * @param file
	 * @return
	 */
	private static String getEntryName(String baseDirPath, File file) {
		/**
		 * 改变 baseDirPath 的形式 把 "C:/temp" 变成 "C:/temp/"
		 */
		if (!baseDirPath.endsWith(File.separator)) {
			baseDirPath += File.separator;
		}
		String filePath = file.getAbsolutePath();
		/**
		 * 测试此抽象路径名表示的文件是否是一个目录。 要是这个文件对象是一个目录 则也要变成 后面带 "/" 这个文件对象类似于
		 * "C:/temp/人体写真/1.jpg" 要是这个文件是一个文件夹 则也要变成 后面带 "/" 因为你要是不这样做,它也会被压缩到目标文件中
		 * 但是却不能正解显示 也就是说操作系统不能正确识别它的文件类型(是文件还是文件夹)
		 */
		if (file.isDirectory()) {
			filePath += "/";
		}
		int index = filePath.indexOf(baseDirPath);
		return filePath.substring(index + baseDirPath.length());
	}

	// //////////////////////////解压缩////////////////////////////////////////
	/**
	 * 调用org.apache.tools.zip实现解压缩，支持目录嵌套和中文名
	 * 也可以使用java.util.zip不过如果是中文的话，解压缩的时候文件名字会是乱码。原因是解压缩软件的编码格式跟java.util.zip.ZipInputStream的编码字符集(固定是UTF-8)不同
	 * 
	 * @param zipFileName     要解压缩的文件
	 * @param outputDirectory 要解压到的目录
	 * @throws Exception
	 */
	public static boolean unZip(String zipFileName, String outputDirectory) {
		boolean flag = false;
		try {
			org.apache.tools.zip.ZipFile zipFile = new org.apache.tools.zip.ZipFile(zipFileName);
			java.util.Enumeration e = zipFile.getEntries();
			org.apache.tools.zip.ZipEntry zipEntry = null;
			createDirectory(outputDirectory, "");
			while (e.hasMoreElements()) {
				zipEntry = (org.apache.tools.zip.ZipEntry) e.nextElement();
				// System.out.println("unziping " + zipEntry.getName());
				if (zipEntry.isDirectory()) {
					String name = zipEntry.getName();
					name = name.substring(0, name.length() - 1);
					File f = new File(outputDirectory + File.separator + name);
					f.mkdir();
					System.out.println("创建目录：" + outputDirectory + File.separator + name);
				} else {
					String fileName = zipEntry.getName();
					fileName = fileName.replace('\\', '/');
					// System.out.println("测试文件1：" +fileName);
					if (fileName.indexOf("/") != -1) {
						createDirectory(outputDirectory, fileName.substring(0, fileName.lastIndexOf("/")));
						fileName = fileName.substring(fileName.lastIndexOf("/") + 1, fileName.length());
					}

					File f = new File(outputDirectory + File.separator + zipEntry.getName());

					f.createNewFile();
					InputStream in = zipFile.getInputStream(zipEntry);
					FileOutputStream out = new FileOutputStream(f);

					byte[] by = new byte[1024];
					int c;
					while ((c = in.read(by)) != -1) {
						out.write(by, 0, c);
					}
					out.close();
					in.close();
				}
				flag = true;
			}
		} catch (Exception ex) {
			ex.printStackTrace();
		}
		return flag;
	}

	/**
	 * 创建目录
	 * 
	 * @param directory    父目录
	 * @param subDirectory 子目录
	 */
	private static void createDirectory(String directory, String subDirectory) {
		String dir[];
		File fl = new File(directory);
		try {
			if (subDirectory == "" && fl.exists() != true)
				fl.mkdir();
			else if (subDirectory != "") {
				dir = subDirectory.replace('\\', '/').split("/");
				for (int i = 0; i < dir.length; i++) {
					File subFile = new File(directory + File.separator + dir[i]);
					if (subFile.exists() == false)
						subFile.mkdir();
					directory += File.separator + dir[i];
				}
			}
		} catch (Exception ex) {
			System.out.println(ex.getMessage());
		}
	}

	// /////////////////////////////////////

	public static void main(String[] temp) {
		// 压缩
		String baseDirName = "C:\\Users\\an\\Desktop\\新建文件夹";
		String[] fileNames = { "红.jpg", "绿.jpg","白.jpg" };
		String zipFileName = "D:\\中文.zip";
		// 压缩多个指定的文件 到ZIP
		System.out.println(TestFile.zip(baseDirName, fileNames, zipFileName, "GBK"));

		// 压缩一个文件夹 到ZIP
		//String sourcePath = "c:\\test\\";
		//String zipFilePath = "c:\\中文2.zip";
		//TestFile.zip(sourcePath, zipFilePath);
		ConcurrentHashMap<String, String> cn = null;
		// 解压缩
		// System.out.println(AntZipUtil.unZip("c:\\中文.zip", "c:\\中文"));
		
	}

}